import pandas as pd
from matplotlib import pyplot as plt

'''
背景：

有两个excel，他们有相同的一个列；
按照这个列合并成一个大的excel，即vlookup功能，要求：
只需要第二个excel的少量的列，比如从40个列中挑选2个列
新增的来自第二个excel的列需要放到第一个excel指定的列后面；
将结果输出到一个新的excel;
'''
def read_excel():
    # 学生成绩表
    df_grade = pd.read_excel("../course_datas/c23_excel_vlookup/学生成绩表.xlsx")
    print(df_grade.head())
    print("--------------------------------------------------")
    # 学生信息表
    df_sinfo = pd.read_excel("../course_datas/c23_excel_vlookup/学生信息表.xlsx")
    print(df_sinfo.head())
    # 目标：怎样将第二个“学生信息表”的姓名、性别两列，添加到第一个表“学生成绩表”，并且放在第一个表的“学号”列后面？
    # 只筛选第二个表的少量的列
    df_sinfo = df_sinfo[["学号", "姓名", "性别"]]
    ss = pd.merge(left=df_grade,right=df_sinfo,left_on="学号",right_on="学号")
    # print(ss)
    # 步骤3：调整列的顺序
    # print(ss.columns)
    new_columns = ss.columns.to_list()

    for name in ['姓名','性别']:
        new_columns.remove(name)
        new_columns.insert(new_columns.index("学号")+1,name)
    print("--------------------------------")
    # 重新 设置 index
    df_merge = ss.reindex(columns = new_columns)
    # print(df_merge)
    df_merge.to_excel("../course_datas/c23_excel_vlookup/合并后的数据表.xlsx", index=False)




if __name__ == '__main__':
    read_excel()

